import os

def split_novel(input_file, output_dir, chunk_size=1000):
    # 创建输出文件夹，如果文件夹不存在
    if not os.path.exists(output_dir):
        os.makedirs(output_dir)
    
    # 读取小说文件
    with open(input_file, 'r', encoding='utf-8') as file:
        content = file.read()

    # 计算总的分块数
    total_chunks = len(content) // chunk_size + (1 if len(content) % chunk_size > 0 else 0)

    # 分割并保存每个文件
    for i in range(total_chunks):
        chunk = content[i * chunk_size:(i + 1) * chunk_size]
        # 使用格式化字符串补充前导零
        output_file = os.path.join(output_dir, f'{i + 1:04}.txt')  # 例如：0001.txt
        
        with open(output_file, 'w', encoding='utf-8') as out_file:
            out_file.write(chunk)

        print(f"File {i + 1:04} saved as {output_file}")

# 使用示例
input_file = '大王饶命.txt'  # 小说文件路径
output_dir = 'sp'         # 存储分割后文件的文件夹

split_novel(input_file, output_dir)
